tensor([[[0.4800, 0.4800],
         [0.4800, 0.4800]],

        [[0.2500, 0.2500],
         [0.2500, 0.2500]],

        [[0.2700, 0.2700],
         [0.2700, 0.2700]]], grad_fn=<RoundBackward1>)